import unittest
from dbtools.mongo import get_mongodb_twitter_connection
from dbtools.mysql import get_connection

class TestDB(unittest.TestCase):

    def setUp(self):
        print('setUp...')

    def tearDown(self):
        print('tearDown...')

    def test_mongodb(self):
        get_mongodb_twitter_connection()

    def test_mysql(self):
        get_connection()


if __name__ == '__main__':
    unittest.main()